Calculator in Python